package com.lishem.top100._05array;

import java.util.Arrays;

/**
 * https://leetcode.cn/problems/product-of-array-except-self/description/
 * <p>
 * 给你一个整数数组 nums，返回 数组 answer ，其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
 * <p>
 * 题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在  32 位 整数范围内。
 * <p>
 * 请不要使用除法，且在 O(n) 时间复杂度内完成此题。
 * <p>
 * 示例 1:
 * <p>
 * 输入: nums = [1,2,3,4]
 * <p>
 * 输出: [24,12,8,6]
 * <p>
 * 示例 2:
 * <p>
 * 输入: nums = [-1,1,0,-3,3]
 * <p>
 * 输出: [0,0,9,0,0]
 */
public class _04LetCode238_除自身以外数组的乘积 {

    public int[] productExceptSelf(int[] nums) {
        int[] res = new int[nums.length];

        // 前缀之积，先暂存在res中
        res[0] = 1;
        for (int i = 1; i < nums.length; i++) {
            res[i] = res[i - 1] * nums[i - 1];
        }

        // 定义变量存储后缀之积
        int tmp = 1;
        for (int i = nums.length - 1; i >= 0; i--) {
            res[i] = res[i] * tmp;
            tmp = tmp * nums[i];
        }
        return res;
    }

    public static void main(String[] args) {
        _04LetCode238_除自身以外数组的乘积 sol = new _04LetCode238_除自身以外数组的乘积();
        System.out.println(Arrays.toString(sol.productExceptSelf(new int[]{1, 2, 3, 4})));
        System.out.println(Arrays.toString(sol.productExceptSelf(new int[]{-1, 1, 0, -3, 3})));
    }
}
